package com.mathmodeling.repository;

import com.mathmodeling.entity.Work;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Repository;

import java.util.List;
import java.util.Optional;

@Repository
public interface WorkRepository extends JpaRepository<Work, String> {
    
    /**
     * 根据团队ID查找作品
     */
    List<Work> findByTeamId(String teamId);
    
    /**
     * 根据竞赛ID查找作品
     */
    List<Work> findByContestId(String contestId);
    
    /**
     * 根据团队ID和竞赛ID查找作品
     */
    Optional<Work> findByTeamIdAndContestId(String teamId, String contestId);
    
    /**
     * 根据题目ID查找作品
     */
    List<Work> findByTopicId(String topicId);
    
    /**
     * 根据竞赛ID和题目ID查找作品
     */
    List<Work> findByContestIdAndTopicId(String contestId, String topicId);
} 